package com.rich.common.util;


import com.rich.common.constant.RandomType;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.concurrent.ThreadLocalRandom;

/**
 * 字符随机生成类
 *
 * @author guofeng
 */
public class RandomString {

    private static final String[] lowercase = {
            "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k",
            "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"};

    private static final String[] capital = {
            "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K",
            "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"};

    private static final String[] number = {
            "1", "2", "3", "4", "5", "6", "7", "8", "9", "0"};

    private static final String[] sign = {
            "~", "!", "@", "#", "$", "%", "^", "&", "*", "(", ")", "_", "+", "`", "-", "=",
            "{", "}", "|", ":", "\"", "<", ">", "?",
            "[", "]", "\\", ";", "'", ",", ".", "/"};

    /**
     * 静态随机数
     */


    public static void main(String[] args) {
        System.out.println(RandomString.getRandom(24, RandomType.LETTER_CAPITAL_NUMBER_SIGN));
    }

    /**
     * 获取随机组合码
     *
     * @param num  位数
     * @param type 类型
     * @RandomType <br>字符型 LETTER,
     * <br>大写字符型 CAPITAL,
     * <br>数字型 NUMBER,
     * <br>符号型 SIGN,
     * <br>大+小字符型 LETTER_CAPITAL,
     * <br>小字符+数字 型 LETTER_NUMBER,
     * <br>大+小字符+数字 型 LETTER_CAPITAL_NUMBER,
     * <br>大+小字符+数字+符号 型 LETTER_CAPITAL_NUMBER_SIGN
     */
    public static String getRandom(int num, RandomType type) {
        ArrayList<String> temp = new ArrayList<String>();
        StringBuffer code = new StringBuffer();
        if (type == RandomType.LETTER) {
            temp.addAll(Arrays.asList(lowercase));
        } else if (type == RandomType.CAPITAL) {
            temp.addAll(Arrays.asList(capital));
        } else if (type == RandomType.NUMBER) {
            temp.addAll(Arrays.asList(number));
        } else if (type == RandomType.SIGN) {
            temp.addAll(Arrays.asList(sign));
        } else if (type == RandomType.LETTER_CAPITAL) {
            temp.addAll(Arrays.asList(lowercase));
            temp.addAll(Arrays.asList(capital));
        } else if (type == RandomType.LETTER_NUMBER) {
            temp.addAll(Arrays.asList(lowercase));
            temp.addAll(Arrays.asList(number));
        } else if (type == RandomType.LETTER_CAPITAL_NUMBER) {
            temp.addAll(Arrays.asList(lowercase));
            temp.addAll(Arrays.asList(capital));
            temp.addAll(Arrays.asList(number));
        } else if (type == RandomType.LETTER_CAPITAL_NUMBER_SIGN) {
            temp.addAll(Arrays.asList(lowercase));
            temp.addAll(Arrays.asList(capital));
            temp.addAll(Arrays.asList(number));
            temp.addAll(Arrays.asList(sign));
        }
        for (int i = 0; i < num; i++) {
            code.append(temp.get(ThreadLocalRandom.current().nextInt(temp.size())));
        }
        return code.toString();
    }

}